from django.http import HttpResponse, JsonResponse
from django.shortcuts import render

import datetime
from user.models import User
from user.models import Ex_info


def index(request):
    return render(request, "index.html")


def crawler(request):
    global sckey
    login_cookie = request.COOKIES.get('login-cookie')
    user_cookie = str(login_cookie).split('/')
    username = user_cookie[0]
    res = User.objects.filter(username=username)
    for i in res:
        sckey = i.SCKEY_ID
    if sckey is not None:
        return render(request, 'crawler.html')
    else:
        return render(request, 'wechat.html')


def wechat(request):
    return render(request, 'wechat.html')


def login(request):
    return render(request, 'login.html')


def city(request):
    city_ = request.POST['city']
    city_dict = {'阿尔山': 'YIE', '阿克苏': 'AKU', '阿拉善右旗': 'RHT', '阿拉善左旗': 'AXF', '阿勒泰': 'AAT', '阿里': 'NGQ', '澳门': 'MFM',
                 '安庆': 'AQG', '安顺': 'AVA', '鞍山': 'AOG', '巴彦淖尔': 'RLK', '百色': 'AEB', '包头': 'BAV', '保山': 'BSD',
                 '北海': 'BHY',
                 '北京': 'BJS', '白城': 'DBC', '白山': 'NBS', '毕节': 'BFJ', '博乐': 'BPL', '重庆': 'CKG', '昌都': 'BPX', '常德': 'CGD',
                 '常州': 'CZX', '朝阳': 'CHG', '成都': 'CTU', '池州': 'JUH', '赤峰': 'CIF', '揭阳': 'SWA', '长春': 'CGQ', '长沙': 'CSX',
                 '长治': 'CIH', '承德': 'CDE', '沧源': 'CWJ', '达县': 'DAX', '大理': 'DLU', '大连': 'DLC', '大庆': 'DQA', '大同': 'DAT',
                 '丹东': 'DDG', '稻城': 'DCY', '东营': 'DOY', '敦煌': 'DNH', '芒市': 'LUM', '额济纳旗': 'EJN', '鄂尔多斯': 'DSN',
                 '恩施': 'ENH',
                 '二连浩特': 'ERL', '佛山': 'FUO', '福州': 'FOC', '抚远': 'FYJ', '阜阳': 'FUG', '赣州': 'KOW', '格尔木': 'GOQ',
                 '固原': 'GYU',
                 '广元': 'GYS', '广州': 'CAN', '贵阳': 'KWE', '桂林': 'KWL', '哈尔滨': 'HRB', '哈密': 'HMI', '海口': 'HAK',
                 '海拉尔': 'HLD',
                 '邯郸': 'HDG', '汉中': 'HZG', '杭州': 'HGH', '合肥': 'HFE', '和田': 'HTN', '黑河': 'HEK', '呼和浩特': 'HET',
                 '淮安': 'HIA',
                 '怀化': 'HJJ', '黄山': 'TXN', '惠州': 'HUZ', '鸡西': 'JXA', '济南': 'TNA', '济宁': 'JNG', '加格达奇': 'JGD',
                 '佳木斯': 'JMU',
                 '嘉峪关': 'JGN', '金昌': 'JIC', '金门': 'KNH', '锦州': 'JNZ', '嘉义': 'CYI', '西双版纳': 'JHG', '建三江': 'JSJ',
                 '晋江': 'JJN',
                 '井冈山': 'JGS', '景德镇': 'JDZ', '九江': 'JIU', '九寨沟': 'JZH', '喀什': 'KHG', '凯里': 'KJH', '康定': 'KGT',
                 '克拉玛依': 'KRY',
                 '库车': 'KCA', '库尔勒': 'KRL', '昆明': 'KMG', '拉萨': 'LXA', '兰州': 'LHW', '黎平': 'HZH', '丽江': 'LJG',
                 '荔波': 'LLB',
                 '连云港': 'LYG', '六盘水': 'LPF', '临汾': 'LFQ', '林芝': 'LZY', '临沧': 'LNJ', '临沂': 'LYI', '柳州': 'LZH',
                 '泸州': 'LZO',
                 '洛阳': 'LYA', '吕梁': 'LLV', '澜沧': 'JMJ', '龙岩': 'LCX', '满洲里': 'NZH', '梅州': 'MXZ', '绵阳': 'MIG',
                 '漠河': 'OHE',
                 '牡丹江': 'MDG', '马祖': 'MFK', '南昌': 'KHN', '南充': 'NAO', '南京': 'NKG', '南宁': 'NNG', '南通': 'NTG',
                 '南阳': 'NNY',
                 '宁波': 'NGB', '宁蒗': 'NLH', '攀枝花': 'PZI', '普洱': 'SYM', '齐齐哈尔': 'NDG', '黔江': 'JIQ', '且末': 'IQM',
                 '秦皇岛': 'BPE',
                 '青岛': 'TAO', '庆阳': 'IQN', '衢州': 'JUZ', '日喀则': 'RKZ', '日照': 'RIZ', '三亚': 'SYX', '厦门': 'XMN',
                 '上海': 'SHA',
                 '深圳': 'SZX', '神农架': 'HPG', '沈阳': 'SHE', '石家庄': 'SJW', '塔城': 'TCG', '台州': 'HYN', '太原': 'TYN',
                 '扬州': 'YTY',
                 '唐山': 'TVS', '腾冲': 'TCZ', '天津': 'TSN', '天水': 'THQ', '通辽': 'TGO', '铜仁': 'TEN', '吐鲁番': 'TLQ',
                 '万州': 'WXN',
                 '威海': 'WEH', '潍坊': 'WEF', '温州': 'WNZ', '文山': 'WNH', '乌海': 'WUA', '乌兰浩特': 'HLH', '乌鲁木齐': 'URC',
                 '无锡': 'WUX',
                 '梧州': 'WUZ', '武汉': 'WUH', '武夷山': 'WUS', '西安': 'SIA', '西昌': 'XIC', '西宁': 'XNN', '锡林浩特': 'XIL',
                 '香格里拉(迪庆)': 'DIG', '襄阳': 'XFN', '兴义': 'ACX', '徐州': 'XUZ', '香港': 'HKG', '烟台': 'YNT', '延安': 'ENY',
                 '延吉': 'YNJ', '盐城': 'YNZ', '伊春': 'LDS', '伊宁': 'YIN', '宜宾': 'YBP', '宜昌': 'YIH', '宜春': 'YIC', '义乌': 'YIW',
                 '银川': 'INC', '永州': 'LLF',
                 '榆林': 'UYN', '玉树': 'YUS', '运城': 'YCU', '湛江': 'ZHA', '张家界': 'DYG', '张家口': 'ZQZ', '张掖': 'YZY',
                 '昭通': 'ZAT',
                 '郑州': 'CGO',
                 '中卫': 'ZHY', '舟山': 'HSN', '珠海': 'ZUH', '遵义(茅台)': 'WMT', '遵义(新舟)': 'ZYI'}
    data = {'msg': 'true'}
    if city_dict.get(city_) is None:
        data['msg'] = 'false'
    return JsonResponse(data)


def submit_city_data(request):
    start_city_ = request.POST['start_city']
    end_city_ = request.POST['end_city']
    start_time_ = request.POST['start_time']
    login_cookie = request.COOKIES.get('login-cookie')
    user_cookie = str(login_cookie).split('/')
    username = user_cookie[0]
    res = User.objects.filter(username=username)
    arr = []
    user_id = 0
    for i in res:
        user_id = int(i.id)
    ex_info = Ex_info(user_id=user_id, start_airport=start_city_, end_airport=end_city_, start_time=start_time_)
    ex_info.save()
    return render(request, 'crawler.html')


def get_ex_info(request):
    login_cookie = request.COOKIES.get('login-cookie')
    user_cookie = str(login_cookie).split('/')
    username = user_cookie[0]
    res = User.objects.filter(username=username)
    arr = []
    user_id = 0
    for i in res:
        user_id = int(i.id)

    res = Ex_info.objects.filter(user_id=user_id)
    arr = []
    for i in res:
        time = i.start_time
        time = time.strftime('%Y/%m/%d')
        content = {'id': i.id, 'user_id': i.user_id, 'start_airport': i.start_airport, 'end_airport': i.end_airport,
                   'start_time': time}
        arr.append(content)

    return JsonResponse({'ex_info': arr})


def delete_ex_info(request):
    id_ = request.GET['id']
    Ex_info.objects.filter(id=id_).delete()
    return render(request, 'crawler.html')
